is prime or not python